#include <stdio.h>
#include <hd_led.h>
#include <hd_oled_protocol_i2c.h>
#include <hd_oled.h>
#include <pd_usart.h>
#include <pd_i2c_hd.h>
#include "main.h"


#define ARRAY_SIZE(x)       sizeof(x) / sizeof(x[0])
#define USE_HARDWARE_SPI    1
#define USE_HARDWARE_I2C    1
uint16_t num;
LED led;
OLED oled;

extern void SystemClock_Config(void);

void oled_instance(OLED *oled);

USART usart;

int main(void) {
    SystemClock_Config();

    NVIC_SetPriorityGrouping(NVIC_PRIORITYGROUP_2);

    oled_instance(&oled);

    OLED_showString(&oled, 0, 0, "USART1 RxData");
    OLED_refresh(&oled);

    GPIO usartGPIO[2];
    GPIO_constructor(usartGPIO + 0, GPIOA, LL_GPIO_PIN_9);
    GPIO_constructor(usartGPIO + 1, GPIOA, LL_GPIO_PIN_10);

    GPIOList uartGPIOList;
    GPIOList_constructor(&uartGPIOList, usartGPIO, 2);

    USART_constructor(&usart, USART1, &uartGPIOList, 9600);
    USART_init(&usart, USART1_IRQn, 1, 1);

    printf("0\r\n");

    while (1) {
        if (USART_getRxFlag(&usart) == 1) {
            const uint8_t RxData = usart.rxData;
            usart.vptr->send_byte(&usart, RxData);
            OLED_printf(&oled, 0, 16, "%c", RxData);
            OLED_refresh(&oled);
        }
    }
}


void USART1_IRQHandler(void) {
    if (LL_USART_IsActiveFlag_PE(USART1) || LL_USART_IsActiveFlag_FE(USART1)
        || LL_USART_IsActiveFlag_ORE(USART1) || LL_USART_IsActiveFlag_NE(USART1)) {
        printf("usart error code: %lX\n", READ_REG(USART1->SR));
    }

    if (LL_USART_IsActiveFlag_RXNE(USART1)) {
        usart.rxData = LL_USART_ReceiveData8(USART1);
        usart.rxFlag = 1;
    } else if (LL_USART_IsActiveFlag_TXE(USART1)) {
    }
}


void oled_instance(OLED *oled) {
    // I2C gpio init
    static GPIO oledSckGPIO, oledSdaGPIO;
    GPIO_constructor(&oledSckGPIO, GPIOB, LL_GPIO_PIN_10); // I2Cx_SCK
    GPIO_constructor(&oledSdaGPIO, GPIOB, LL_GPIO_PIN_11); // I2Cx_SDA

#ifdef USE_HARDWARE_I2C
    static I2CHardware i2c2;
    I2CHardware_constructor(&i2c2, I2C2, &oledSckGPIO, &oledSdaGPIO);
#elif USE_SIMULATE_I2C
    static I2CSimulate i2c2;
    I2CSimulate_constructor(&i2c2, &oledSckGPIO, &oledSdaGPIO);
#endif
    I2C_init((I2C *)&i2c2);

    // OLED protocol of I2C
    static OLEDProtocolI2C oledProtocolI2c;
    OLEDProtocolI2C_constructor(&oledProtocolI2c, (I2C *)&i2c2);

    // OLED font instance.
    static OLEDFont oledFont;
    OLEDFont_contructor(&oledFont, NULL);

    // OLED instance of 4-PIN(I2C)
    OLED_constructor(oled, (OLEDProtocol *)&oledProtocolI2c, &oledFont);

    OLED_init(oled);

    OLED_clear(oled);
    OLED_refresh(oled);
}
